package cn.edu.scau.cmi.ema.controller.rest;

import org.springframework.stereotype.Controller;

import java.util.List;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;
import java.io.IOException;

import com.google.gson.Gson;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;


import cn.edu.scau.cmi.ema.base.CmiPagination;
import cn.edu.scau.cmi.ema.service.base.CmiPsonService;
import cn.edu.scau.cmi.ema.base.CmiEmaController;
import cn.edu.scau.cmi.ema.dao.*;
import cn.edu.scau.cmi.ema.domain.*;
import cn.edu.scau.cmi.ema.service.*;

import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletRequest;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.bind.WebDataBinder;

@Controller("SiteRestController")
public class SiteRestController extends CmiEmaController {

	/**
	 * Create a new Site entity
	 * 
	 */
	@RequestMapping(value = "/newSite", method = RequestMethod.POST)
	@ResponseBody
	public Object newSite(@RequestBody Site site) {
		siteService.saveSite(site);
		return JSONObject.parse(cmiPsonService.toJsonString(siteDAO.findSiteByPrimaryKey(site.getId()), 2));
	}

	@RequestMapping(value = "/restNewSite", method = RequestMethod.POST)
	@ResponseBody
	public Object restNewSite(@RequestBody String siteJson) {
		Site site = new Gson().fromJson(siteJson, Site.class);
		appendReferences(site);
		siteService.saveSite(site);
		return JSONObject.parse(cmiPsonService.toJsonString(siteDAO.findSiteByPrimaryKey(site.getId()), 2));
	}

	@RequestMapping(value = "/saveSite", method = RequestMethod.PUT)
	@ResponseBody
	public Site saveSite(@RequestBody Site site) {
		siteService.saveSite(site);
		return siteDAO.findSiteByPrimaryKey(site.getId());
	}

	@RequestMapping(value = "/restSaveSite", method = RequestMethod.PUT)
	@ResponseBody
	public Object restSaveSite(@RequestBody String siteJson) {
		Site site = new Gson().fromJson(siteJson, Site.class);
		appendReferences(site);
		CmiPsonService cmiPsonService = new CmiPsonService();
		siteService.saveSite(site);
		return JSONObject.parse(cmiPsonService.toJsonString(siteDAO.findSiteByPrimaryKey(site.getId()), 1));
	}

	@RequestMapping(value = "/restDeleteSite/{siteId}", method = RequestMethod.DELETE)
	@ResponseBody
	public void restDeleteSite(@PathVariable Integer siteId) {
		Site site = siteDAO.findSiteByPrimaryKey(siteId);
		siteService.deleteSite(site);
	}

	@RequestMapping(value = "/restConfirmDeleteSites")
	@ResponseBody
	public Object restConfirmDeleteSites(@RequestBody Set<Integer> deleteSitesIds) {
		Set<Site> deleteSites = siteDAO.findSitesByIds(deleteSitesIds);
		Set<Site> referencedSites = siteService.loadReferenceSites(deleteSites);
		deleteSites.removeAll(referencedSites);
		JSONObject jsonObject = new JSONObject();
		jsonObject.put("referencedEntitys", referencedSites);
		jsonObject.put("deleteEntitys", deleteSites);
		return JSONObject.parse(jsonObject.toJSONString());
	}

	@RequestMapping("/restDeleteSites")
	@ResponseBody
	public Object restDeleteSites(@RequestBody Set<Site> deleteSites) {
		Iterator<Site> iterator = deleteSites.iterator();
		while (iterator.hasNext()) {
			Site currentSite = iterator.next();
			siteService.deleteSite(currentSite);
		}
		JSONObject jsonObject = new JSONObject();
		jsonObject.put("url", "listSites");
		return JSONObject.parse(jsonObject.toJSONString());
	}
	// loadself

	@RequestMapping(value = "/restLoadSite/{siteId}", method = RequestMethod.POST)
	@ResponseBody
	public Object restLoadSite(@PathVariable Integer siteId, @RequestBody CmiPagination cmiPagination) {
		CmiPsonService cmiPsonService = new CmiPsonService();
		return JSONObject.parse(cmiPsonService.toPsonString(siteDAO.findSiteByPrimaryKey(siteId), 2, cmiPagination));
	}
	// loadReferenceds
	// TODO loadReferencedsPson，这个方法有一些问题，待修复

	@RequestMapping(value = "/restLoadSiteRules/{siteId}/rules/{relatedruleId}", method = RequestMethod.POST)
	@ResponseBody
	public Object restLoadSiteRules(@PathVariable Integer siteId, @PathVariable Integer relatedruleId,
			@RequestBody CmiPagination cmiPagination) {
		Rule rule = ruleDAO.findRuleByPrimaryKey(relatedruleId, -1, -1);
		CmiPsonService cmiPsonService = new CmiPsonService();
		return JSONObject.parse(cmiPsonService.toPsonString(rule, 2, cmiPagination));
	}

	@RequestMapping(value = "/restLoadSiteUrls/{siteId}/urls/{relatedurlId}", method = RequestMethod.POST)
	@ResponseBody
	public Object restLoadSiteUrls(@PathVariable Integer siteId, @PathVariable Integer relatedurlId,
			@RequestBody CmiPagination cmiPagination) {
		Url url = urlDAO.findUrlByPrimaryKey(relatedurlId, -1, -1);
		CmiPsonService cmiPsonService = new CmiPsonService();
		return JSONObject.parse(cmiPsonService.toPsonString(url, 2, cmiPagination));
	}

	public void appendReferences(Site site) {

		// 引用表不是纯联系表，引用数量大于或者等于2
		Set<Rule> rulesForSite = new HashSet<>();
		for (Rule rule_Site : site.getRulesForSite()) {
			rulesForSite.add(ruleService.findRuleByPrimaryKey(rule_Site.getId()));
		}
		site.setRulesForSite(rulesForSite);

		// 引用表不是纯联系表，引用数量大于或者等于2
		Set<Url> urlsForSite = new HashSet<>();
		for (Url url_Site : site.getUrlsForSite()) {
			urlsForSite.add(urlService.findUrlByPrimaryKey(url_Site.getId()));
		}
		site.setUrlsForSite(urlsForSite);
	}

	@RequestMapping(value = "/restGetSiteRules/{siteId}/rules", method = RequestMethod.POST)
	@ResponseBody
	public Object restGetSiteRules(@PathVariable Integer siteId, @RequestBody CmiPagination cmiPagination) {
		CmiPsonService cmiPsonService = new CmiPsonService();
		return JSONObject
				.parse(cmiPsonService.toPsonString(siteDAO.findSiteByPrimaryKey(siteId).getRulesForSite(), 2, cmiPagination));
	}

	@RequestMapping(value = "/restGetSiteUrls/{siteId}/urls", method = RequestMethod.POST)
	@ResponseBody
	public Object restGetSiteUrls(@PathVariable Integer siteId, @RequestBody CmiPagination cmiPagination) {
		CmiPsonService cmiPsonService = new CmiPsonService();
		return JSONObject
				.parse(cmiPsonService.toPsonString(siteDAO.findSiteByPrimaryKey(siteId).getUrlsForSite(), 2, cmiPagination));
	}

	@RequestMapping(value = "/restListSites", method = RequestMethod.GET)
	@ResponseBody
	public Object restListSites() {
		return JSONObject.parse(cmiPsonService.toJsonString(siteService.loadSites(), 2));
	}

	@RequestMapping(value = "/restListSitesPost", method = RequestMethod.POST)
	@ResponseBody
	public Object restListSitesPost(@RequestBody CmiPagination cmiPagination) {

		CmiPsonService cmiPsonService = new CmiPsonService();
		cmiPagination = sitePaginationService.updateAppPagination(cmiPagination);
		int startResult = (cmiPagination.getCurrentPagination() - 1) * cmiPagination.getPaginationSize();

		Set<Site> sites = siteDAO.findAllSites(startResult, cmiPagination.getPaginationSize());
		String siteString = cmiPsonService.toPsonString(sites, 2, cmiPagination);

		return JSONObject.parse(siteString);
	}

	@RequestMapping("/siteSelectBox")
	@ResponseBody
	public JSONObject siteSelectBox(@RequestParam(required = false) String searchTable,
			@RequestParam(required = false) Set<Integer> searchValue, // 杨宏宇2019-9-10：searchValue是在selectpage.js的616行定义，如果要修改searchValue的名称，1.可以修改在selectpage.js修改；2.可以
																		// @RequestParam(value="searchValue", required =
																		// false) Set<Integer> someName
			@RequestParam(required = false, defaultValue = "10") int paginationSize,
			@RequestParam(required = false, defaultValue = "1") int pageNumber,
			@RequestParam(value = "qWord[]", required = false) String queryword,
			@RequestParam(required = false) String selectItem) throws IOException {
		Set entitySet = null;
		Integer count = 0;

		// 初始化原始值

		if (searchValue != null) {
			switch (searchTable) {
			// 外键

			// 被引用键
			case "Rule":
				entitySet = ruleService.loadRulesByIdSet(searchValue);
				break;
			case "Url":
				entitySet = urlService.loadUrlsByIdSet(searchValue);
				break;
			}
			// 没有测试，深度需要多少？？？
			String jsonData = cmiPsonService.toJsonString(entitySet, 1);
			JSONArray jsonArray = new JSONArray();
			jsonArray = (JSONArray) JSON.parse((String) jsonData);

			JSONObject jsonObject = new JSONObject();
			jsonObject.put("list", jsonArray);
			jsonObject.put("totalRow", entitySet.size());
			return jsonObject;
		}

		// 不根据关键字进行搜索
		if (queryword == null || queryword.trim() == "") {
			switch (selectItem) {
			// 外键

			// 被引用键
			case "Rule":
				entitySet = ruleDAO.findAllRules((pageNumber - 1) * paginationSize, paginationSize);
				count = ruleService.countRules();
				break;
			case "Url":
				entitySet = urlDAO.findAllUrls((pageNumber - 1) * paginationSize, paginationSize);
				count = urlService.countUrls();
				break;
			}
		} else {
			switch (selectItem) {
			// 外键

			// 被引用键
			case "Rule":
				entitySet = ruleDAO.findRuleByNameContaining(queryword, (pageNumber - 1) * paginationSize,
						paginationSize);
				count = ruleService.countRulesByName(queryword);
				break;
			case "Url":
				entitySet = urlDAO.findUrlByNameContaining(queryword, (pageNumber - 1) * paginationSize,
						paginationSize);
				count = urlService.countUrlsByName(queryword);
				break;
			}
		}
		/**
		 * 将Json数据转换成jsonArray形式，需要通过pson方式进行转换，然后再转换成JsonArray。因为存在懒加载
		 * 没有直接使用CmiPsonService中的fromJson方法是因为，fromJson方法会将int类型的直接转换成double类型的数据，这个问题还没有解决
		 */
		// 没有测试，深度需要多少？？？
		String jsonData = cmiPsonService.toJsonString(entitySet, 1);
		JSONArray jsonArray = new JSONArray();
		jsonArray = (JSONArray) JSON.parse((String) jsonData);

		JSONObject jsonObject = new JSONObject();
		jsonObject.put("list", jsonArray);
		jsonObject.put("paginationSize", paginationSize);
		jsonObject.put("pageNumber", pageNumber);
		jsonObject.put("totalRow", count);
		jsonObject.put("totalPaginations",
				count % paginationSize == 0 ? count / paginationSize : count / paginationSize + 1);
		return jsonObject;
	}
}
